from sqlalchemy import Column, Integer, String, Text, DateTime, JSON, Date, Float, Index, DECIMAL
from sqlalchemy.sql import func
from db.database import Base
import uuid


class FamilyTreeModel(Base):
    """族谱模型 - 存储家族成员信息和关系"""
    __tablename__ = "family_tree"

    # 主键，使用UUID
    id = Column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    
    # 基本信息字段
    name = Column(String(50), nullable=False, comment="姓名 - 必填字段")
    taboo_name = Column(String(50), nullable=True, comment="名讳")
    ranking_character = Column(String(20), nullable=True, comment="排行")
    birth_date = Column(Date, nullable=True, comment="出生日期")
    gender = Column(Integer, default=0, comment="性别 0-未知 1-男 2-女")
    avatar = Column(String(500), nullable=True, comment="头像URL")
    
    # 族谱关系字段
    generation = Column(Integer, nullable=True, comment="第几代")
    
    # 生平和其他信息
    biography = Column(Text, nullable=True, comment="生平简介")
    death_date = Column(Date, nullable=True, comment="死亡时间")
    
    # 墓地相关信息
    longitude = Column(DECIMAL(10, 6), nullable=True, comment="墓地经度坐标")
    latitude = Column(DECIMAL(10, 6), nullable=True, comment="墓地纬度坐标")
    grave_address = Column(String(500), nullable=True, comment="墓地地址名称")
    grave_images = Column(JSON, nullable=True, comment="墓地周边图片URL数组")
    
    # 其他图片
    images = Column(JSON, nullable=True, comment="相关图片URL数组")
    
    # 多租户和多店铺字段
    shop_id = Column(String(36), nullable=True, comment="店铺ID")
    tenant_id = Column(String(36), nullable=True, comment="租户ID")
    
    # 系统字段
    create_by = Column(String(36), nullable=True, comment="创建人ID")
    update_by = Column(String(36), nullable=True, comment="修改人ID")
    create_time = Column(DateTime(timezone=True), server_default=func.now(), comment="创建时间")
    update_time = Column(DateTime(timezone=True), server_default=func.now(), onupdate=func.now(), comment="修改时间")
    status = Column(Integer, default=1, comment="状态 1-正常 0-删除")
    deleted_at = Column(DateTime(timezone=True), nullable=True, comment="删除时间")
    
    # 添加索引以优化查询性能
    __table_args__ = (
        # 基础索引（保持）
        Index('idx_name_status', 'name', 'status'),
        
        # 新增优化索引
        # 优化多租户代数查询性能（POST /members-generation/{generation}）
        Index('idx_generation_tenant_shop_status', 'generation', 'tenant_id', 'shop_id', 'status'),
        # 优化多租户族谱结构查询性能（POST /structure）
        Index('idx_tenant_shop_status_id', 'tenant_id', 'shop_id', 'status', 'id'),
        # 优化姓名搜索 + 多租户筛选性能（GET /members-search）
        Index('idx_name_tenant_shop_status', 'name', 'tenant_id', 'shop_id', 'status'),
        # 优化墓地地理位置查询性能（POST /graves-nearby）
        Index('idx_location_tenant_shop_status', 'longitude', 'latitude', 'tenant_id', 'shop_id', 'status'),
    )


class FamilyRelationshipModel(Base):
    """家族关系模型 - 存储家族成员之间的关系"""
    __tablename__ = "family_relationship"

    # 主键
    id = Column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    
    # 关系双方 - 使用普通字符串字段，不使用外键约束
    person_id = Column(String(36), nullable=False, comment="人员ID")
    person_name = Column(String(50), nullable=True, comment="人员姓名（缓存字段）")
    related_person_id = Column(String(36), nullable=False, comment="关系人ID")
    related_person_name = Column(String(50), nullable=True, comment="关系人姓名（缓存字段）")
    
    # 关系类型和角色
    relationship_type = Column(String(30), nullable=False, comment="关系类型: parent_child, spouse, sibling, grandparent_grandchild")
    person_role = Column(String(30), nullable=False, comment="person_id的角色: father, mother, son, daughter, spouse, brother, sister, elder_brother, younger_brother, elder_sister, younger_sister, grandfather, grandmother, grandson, granddaughter")
    related_person_role = Column(String(30), nullable=False, comment="related_person_id的角色: father, mother, son, daughter, spouse, brother, sister, elder_brother, younger_brother, elder_sister, younger_sister, grandfather, grandmother, grandson, granddaughter")
    
    # 多租户和多店铺字段
    shop_id = Column(String(36), nullable=True, comment="店铺ID")
    tenant_id = Column(String(36), nullable=True, comment="租户ID")
    
    # 系统字段
    create_by = Column(String(36), nullable=True, comment="创建人ID")
    create_time = Column(DateTime(timezone=True), server_default=func.now(), comment="创建时间")
    update_time = Column(DateTime(timezone=True), server_default=func.now(), onupdate=func.now(), comment="修改时间")
    status = Column(Integer, default=1, comment="状态 1-正常 0-删除")
    
    # 添加索引以优化查询性能
    __table_args__ = (
        # 基础索引（保持）
        Index('idx_person_related_status', 'person_id', 'related_person_id', 'status'),
        Index('idx_unique_relationship', 'person_id', 'related_person_id', 'relationship_type', 'person_role', 'related_person_role', 'status'),
        Index('idx_person_shop_status', 'person_id', 'shop_id', 'status'),
        
        # 新增核心优化索引（祖先/后代递归查询）
        # 优化祖先查询性能（双向 + 多租户）
        Index('idx_person_tenant_shop_status', 'person_id', 'tenant_id', 'shop_id', 'status'),
        Index('idx_related_tenant_shop_status', 'related_person_id', 'tenant_id', 'shop_id', 'status'),
        # 优化关系类型 + 多租户查询性能
        Index('idx_type_tenant_shop_status', 'relationship_type', 'tenant_id', 'shop_id', 'status'),
        # 优化祖先/后代递归查询性能（关键索引）
        Index('idx_person_type_tenant_shop', 'person_id', 'relationship_type', 'tenant_id', 'shop_id', 'status'),
        Index('idx_related_type_tenant_shop', 'related_person_id', 'relationship_type', 'tenant_id', 'shop_id', 'status'),
    )